package com.holic.spring.aigc.model.spark;

import com.fasterxml.jackson.annotation.JsonProperty;
import lombok.Builder;
import lombok.Data;

import java.util.List;

/**
 * @ClassName SparkChatRequest
 * @Description
 * @Author Neil
 * @Date 2024/11/19 17:25
 * @Version V1.0
 **/
@Data
@Builder
public class SparkRequest {

    @Builder.Default
    private String model = "lite";
    @Builder.Default
    private String user = "user";
    private List<SparkMessage> messages;
    @Builder.Default
    @JsonProperty("max_tokens")
    private Integer maxTokens = 4096;
    @Builder.Default
    @JsonProperty("top_k")
    private Integer topK =4;
    @Builder.Default
    private Float temperature = 0.5f;
    @Builder.Default
    private Boolean stream = false;
    @Builder.Default
    @JsonProperty("presence_penalty")
    private Integer presencePenalty = 1;
    @Builder.Default
    @JsonProperty("frequency_penalty")
    private Integer frequencyPenalty = 1;

}
